Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jit compile #1107

Open
wants to merge 18 commits into
base: main
Choose a base branch
from
Open

Jit compile #1107

wants to merge 18 commits into from

Conversation

calad0i
Copy link
Contributor

@calad0i calad0i commented Oct 30, 2024

Description

Allow jit compiling vivado/vitis (and quartus, but sometimes slower) cpp code. Compilation time may be reduced or increased, depending on the exact configuration (reduced if weights are not embedded in header in general, and vice versa).

Unfortunately, due to some confliction (probably related to macros), one isolated process is required for each jitted-model to separate the runtime. This introduces some overhead in runtime and ram usage.

Setting HLS4ML_USE_JIT=1 in envvar will enable jit compilation for supported models by default.

Setting EXTRA_CLING_ARGS=-fopenmp will enable parallel inference, if openmp presents.

Type of change

  • New feature (non-breaking change which adds functionality)

Tests

Run the current set of tests with HLS4ML_USE_JIT=1 will work. Added test_jit.py for standlone test.

Checklist

  • I have read the guidelines for contributing.
  • I have commented my code, particularly in hard-to-understand areas.
  • I have made corresponding changes to the documentation.
  • My changes generate no new warnings.
  • I have installed and run pre-commit on the files I edited or added.
  • I have added tests that prove my fix is effective or that my feature works.

@calad0i calad0i added the please test Trigger testing by creating local PR branch label Oct 30, 2024
@calad0i
Copy link
Contributor Author

calad0i commented Oct 31, 2024

pre-commit.ci autofix

@calad0i calad0i added please test Trigger testing by creating local PR branch and removed please test Trigger testing by creating local PR branch labels Oct 31, 2024
@calad0i calad0i added please test Trigger testing by creating local PR branch and removed please test Trigger testing by creating local PR branch labels Oct 31, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
please test Trigger testing by creating local PR branch
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant